# 作者：位微
# 你不是在写代码而是在解决问题
# 2025年09月17日17时12分26秒
# 2062873410@qq.com
import torch
import numpy as np

a=torch.zeros(2,4)
b=torch.ones(2,4)
# 使用更多
c=torch.cat((a,b),dim=0)#沿0轴进行拼接·，零轴发生改变

e=np.zeros((2,4),np.int32)
f=np.ones((2,4),np.int32)
g=np.concat((e,f),axis=0)
print(g)
print(a.shape)
print(c.shape)
print(c)
print('-'*50)
a=torch.linspace(1,9,9).view(3,3)
b=torch.linspace(1,9,9).view(3,3)
print('a',a)
print('b',b)
# print(a.shape)
c=torch.stack((a,b),dim=0)# 这个是直接加了个0轴
d=torch.stack((a,b),dim=1)
# print('d[:,0,:]',d[:,0,:]) # 这里取1轴的第0个得到了a,因为这里是按照1轴进行拼接的
# print('d[:,1,:]',d[:,1,:])# 对应b
# print('d',d.shape)
# print('c',c.shape)
print('c',c)
print('d',d)

